Naive bayes classifier in python